choice function in python